# model_loader.py
from paddlelite.lite import *

class ModelLoader:
    @staticmethod
    def load_model(model_dir):
        print("开始加载模型....")
        config = MobileConfig()
        config.set_model_from_file(model_dir)
        predictor = create_paddle_predictor(config)
        print("模型加载完毕....")
        return predictor
